import numpy as np
import tensorflow as tf
from tensorflow import keras
from policy_model import Policy

data = np.load('database.npz')
X = data['X']
Y = data['Y']

policy = Policy(dropout=True)
policy.fit(X, Y, 0.05)

policy.save('policy_test.h5')